package demo4.mapper;

import demo4.pojo.User;
import demo4.util.IDBUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.ArrayList;
import java.util.List;

@Component
public class UserMapper implements IUserMapper{
    @Autowired
    private IDBUtil dbUtil;
    @Override
    public int insert(User user) {
        int result = 0;
        try (Connection connection = dbUtil.getConnection()) {
            String sql = "insert into  user (username,password,nickname,age) values(?,?,?,?)";
            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setString(1,user.getUsername());
            ps.setString(2,user.getPassword());
            ps.setString(3,user.getNickname());
            ps.setInt(4,user.getAge());
            result = ps.executeUpdate();
        }catch (Exception e){
            e.printStackTrace();
        }
        return result;
    }

    public int select(User user){
        int flag = 0;
        try(Connection connection = dbUtil.getConnection()){
            String sql = "select username,password from user";
            PreparedStatement ps = connection.prepareStatement(sql);
            ResultSet rs = ps.executeQuery();
            while (rs.next()){
                String username = rs.getString("username");
                String password = rs.getString("password");
                if (username.equals(user.getUsername()) && password.equals(user.getPassword())){
                    flag = 1;
                }
            }
        }catch (Exception e){
            e.printStackTrace();
        }

        return flag;
    }



    @Override
    public User selectByUsername(String username) {
        try (Connection connection = dbUtil.getConnection()) {
            String sql = "select id,username,password,nickname,age from user where username = ?";
            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setString(1,username);
            ResultSet rs = ps.executeQuery();
            while (rs.next()){
                User user = new User();
                user.setId(rs.getInt("id"));
                user.setUsername(rs.getString("username"));
                user.setPassword(rs.getString("password"));
                user.setNickname(rs.getString("nickname"));
                user.setAge(rs.getInt("age"));
                return user;
            }
        }catch (Exception e){
            e.printStackTrace();
        }
        return null;
    }



    @Override
    public List<User> selectAll() {
        List< User> list = new ArrayList<>();
        try (Connection connection = dbUtil.getConnection()) {
            String sql = "select * from user";
            PreparedStatement ps = connection.prepareStatement(sql);
            ResultSet rs = ps.executeQuery();
            while (rs.next()){
                User user = new User();
                user.setUsername(rs.getString("username"));
                user.setPassword(rs.getString("password"));
                user.setNickname(rs.getString("nickname"));
                user.setAge(rs.getInt("age"));
                list.add(user);
            }
        }catch (Exception e){
        }
        return list;
    }

    @Override
    public int userUpdate(User user) {
        try (Connection connection = dbUtil.getConnection()) {
            String sql = "update user set password = ?,nickname = ?,age = ? where username = ? ";
            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setString(1,user.getPassword());
            ps.setString(2,user.getNickname());
            ps.setInt(3,user.getAge());
            ps.setString(4,user.getUsername());
            return ps.executeUpdate();
        }catch (Exception e){
            e.printStackTrace();
        }

        return 0;
    }

}
